import torch
import numpy as np
import array

import torch

input = torch.arange(16).reshape(4,4)
outpu=torch.select(input, 0, 0) 
print(outpu)